#pragma once
#include <iostream>
#include <semaphore.h>
#include <vector>

enum
{
    DEFAULT,
    PTHREAD_CREATE_FAIL,
};
template <class T>
class RoundQueue
{
    const static int defaultsize = 5;

private:
    void P(sem_t &p_sem)
    {
        sem_wait(&p_sem);
    }
    void V(sem_t &c_sem)
    {
        sem_post(&c_sem);
    }
    void Lock(pthread_mutex_t& lock)
    {
        pthread_mutex_lock(&lock);
    }
    void UnLock(pthread_mutex_t& lock)
    {
        pthread_mutex_unlock(&lock);
    }

public:
    RoundQueue(int cap = defaultsize)
        : _rq(cap), _p_index(0), _c_index(0), _cap(cap)
    {
        sem_init(&_c_sem, 0, 0);
        sem_init(&_p_sem, 0, cap);
        pthread_mutex_init(&_p_mutex,nullptr);
        pthread_mutex_init(&_c_mutex,nullptr);
    }
    void push(const T &in) // 生产
    {
        P(_p_sem);
        Lock(_p_mutex);
        _rq[_p_index] = in;
        // 位置后移，维持环形特性
        _p_index++;
        _p_index %= _cap;
        UnLock(_p_mutex);
        V(_c_sem);
    }
    T pop()       // 消费
    {
        P(_c_sem);
        Lock(_c_mutex);

        T out = _rq[_c_index];
        // 位置后移，维持环形特性
        _c_index++;
        _c_index %= _cap;
        UnLock(_c_mutex);

        V(_p_sem);
        return out;
    }

    ~RoundQueue() 
    {
        sem_destroy(&_c_sem);
        sem_destroy(&_p_sem);
        pthread_mutex_destroy(&_p_mutex);
        pthread_mutex_destroy(&_c_mutex);
    };

private:
    std::vector<T> _rq;
    // 下标
    int _p_index;
    int _c_index;
    // 容量
    int _cap;
    // 锁
    pthread_mutex_t _p_mutex;
    pthread_mutex_t _c_mutex;

    // 信号量
    sem_t _c_sem;
    sem_t _p_sem;
};